# -*- coding: utf-8 -*-
# -------------------------------
# @项目：机器视觉模型
# @文件：liaocheng_diaoyong.py
# @时间：2024/4/3 16:19
# @作者：AnFany
# -------------------------------
# 不同站点的配置文件不同
import argparse
import jietu_shuimian as jit

parser = argparse.ArgumentParser(description="污水识别")

# parser.add_argument("--orifig", help="站点的配置图片", required=True)
# parser.add_argument("--labelcsv",  help="站点的配置csv", required=True)
# parser.add_argument("--testfig",  help="需要检验的图片", required=True)
orifig = 'models/sewage_detection/09759838902925950101#1185f4f6bb2e4941b9b954198376e30c/ori.jpg'
labelcsv = 'models/sewage_detection/09759838902925950101#1185f4f6bb2e4941b9b954198376e30c/label.csv'
config = 'models/sewage_detection/09759838902925950101#1185f4f6bb2e4941b9b954198376e30c/config_CNN.ini'
testfig = 'models/sewage_detection/09759838902925950101#1185f4f6bb2e4941b9b954198376e30c/test.jpg'



args = parser.parse_args()
# 返回结果
#result = jit.SX(args.orifig, args.labelcsv, args.testfig)
result = jit.SX(orifig, labelcsv, config, testfig)
print(result.select_fig())


